// +----------------------------------------------------------------------
// | CRMEB [ CRMEB赋能开发者，助力企业发展 ]
// +----------------------------------------------------------------------
// | Copyright (c) 2016~2022 https://www.crmeb.com All rights reserved.
// +----------------------------------------------------------------------
// | Licensed CRMEB并不是自由软件，未经许可不能去掉CRMEB相关版权
// +----------------------------------------------------------------------
// | Author: CRMEB Team <admin@crmeb.com>
// +----------------------------------------------------------------------

import config from '@/config';
import axios from 'axios';

// 创建axios实例
const deepseekApi = axios.create({
  baseURL: config.deepseek.baseUrl,
  headers: {
    'Authorization': `Bearer ${config.deepseek.apiKey}`,
    'Content-Type': 'application/json',
  },
});

/**
 * @description 生成图谱数据
 * @param {Object} param data {Object} 传值参数
 */
export function generateGraphDataApi(data) {
  return deepseekApi({
    url: '/chat/completions',
    method: 'post',
    data: {
      model: 'deepseek-chat',
      messages: [
        {
          role: 'system',
          content: '你是一个专业的知识图谱构建师，你的任务是根据用户提供的主题生成一个结构化的知识图谱。图谱应该以JSON格式返回，包含节点和关系，并且节点应该有层级结构，可以用于树状图展示。'
        },
        {
          role: 'user',
          content: `请根据以下主题生成一个知识图谱：${data.topic}`
        }
      ],
      response_format: { type: 'json_object' }
    },
  });
}

export default deepseekApi;